package com.heima.media.service.impl;

import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.heima.article.dto.ArticleDto;
import com.heima.article.feign.ArticleFeign;
import com.heima.common.aliyun.service.AliyunGreenService;
import com.heima.common.dto.ResponseResult;
import com.heima.common.enums.AppHttpCodeEnum;
import com.heima.common.enums.MediaEnum;
import com.heima.common.minio.service.MinIOService;
import com.heima.common.util.SensitiveWordUtil;
import com.heima.media.dto.ContentDto;
import com.heima.media.entity.WmChannel;
import com.heima.media.entity.WmNews;
import com.heima.media.entity.WmSensitive;
import com.heima.media.entity.WmUser;
import com.heima.media.service.*;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang3.StringUtils;
import org.redisson.api.RBlockingDeque;
import org.redisson.api.RDelayedQueue;
import org.redisson.api.RedissonClient;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import java.io.InputStream;
import java.util.*;
import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;

import static com.heima.common.enums.MediaEnum.NEWS_SUBMIT_STATUS_ARTIFICIAL;
import static com.heima.common.enums.MediaEnum.NEWS_SUBMIT_STATUS_FAIL;

/**
 * 自媒体文章审核业务类
 *
 * @author: itheima
 * @create: 2021-12-25 10:13
 */
@Service
public class AuditServiceImpl implements IAuditService {

    @Autowired
    private IWmNewsService newsService;

    @Autowired
    private ArticleFeign articleFeign;

    @Autowired
    private IWmUserService wmUserService;

    @Autowired
    private IWmChannelService channelService;

    @Autowired
    private MinIOService minIOService;

    @Autowired
    private AliyunGreenService aliyunGreenService;


    @Autowired
    private RedissonClient redissonClient;


    /**
     * 根据入参中自媒体文章ID审核文章
     *
     * @param id 自媒体文章ID
     */
    @Override
    @Async
    public void audit(Integer id) {
        //1.根据自媒体文章ID查询自媒体对象
        WmNews wmNews = newsService.getById(id);
        //2.调用审核自媒体文章方法
        audit(wmNews);
    }

    /**
     * 完成对自媒体文章审核（文本，图片）
     *
     * @param wmNews 自媒体文章
     */
    @Override
    public void audit(WmNews wmNews) {
        //1.自动审核通过待发布/人工审核通过的文章
        Integer status = wmNews.getStatus();
        //1.1 得到文章目前状态，根据文章发布时间是否大于当前时间决定是否将文章进行发布（远程调用文章微服务）
        if (status.equals(MediaEnum.NEWS_SUBMIT_STATUS_ARTIFICIAL_PASS.getVal()) || status.equals(MediaEnum.NEWS_SUBMIT_STATUS_PASS.getVal())) {
            if (System.currentTimeMillis() >= wmNews.getPublishTime().getTime()) {
                //1.2 远程调用文章微服务发布文章到文章库中
                remoteSaveArticle(wmNews);
                return;
            }
        }
        //2.处理自媒体文章状态为“待审核”，进入自动审核业务流程中 TODO:检测文章中是否包含自定义敏感词
        if (status.equals(MediaEnum.NEWS_SUBMIT_STATUS_COMMIT.getVal())) {
            //2.0 从自媒体文章对象中获取文章内容跟图片
            Map<String, Object> map = getTextAndImageFromContent(wmNews.getContent());
            //2.1 审核文章文本（文章标题，文章正文中文本）
            String contentText = (String) map.get("text");
            String text = wmNews.getTitle() + contentText;

            //TODO 采用自定义敏感词过滤
            boolean b = checkSensitive(text, wmNews);
            if (!b) {
                throw new RuntimeException("文章中包含敏感词审核失败！");
            }
            boolean textResult = checkText(text, wmNews);
            if (!textResult) {
                return;
            }

            //2.2 审核文章图片（文章封面，文章正文中图片）
            List<String> images = (List<String>) map.get("images");
            //2.2.1 获取文章封面图片
            String converImages = wmNews.getImages();
            if (StringUtils.isNotBlank(converImages)) {
                String[] split = converImages.split(",");
                List<String> strings = Arrays.asList(split);
                //将封面图片集合添加到所有待检测图片集合中
                images.addAll(strings);
            }
            boolean imgaesResult = checkImage(images, wmNews);
            if (!imgaesResult) {
                return;
            }
            //2.3 审核自媒体文章（两种情况：1未到发布时间：修改状态 2已经到发布时间） 通过远程调用文章微服务完成文章发布
            //判断逻辑：文章发布时间跟当前时间进行比较。如果发布时间大于当前时间，不是直接发布文章，而是需要等到定义发布时间再去发布
            if (wmNews.getPublishTime() == null || wmNews.getPublishTime().getTime() <= System.currentTimeMillis()) {
                //2.3.1 未指定发布时间 或者 发布时间小于当前时间----》》》直接进行发布
                remoteSaveArticle(wmNews);
            } else {
                //2.3.2 自媒体文章审核通过，但是没有到发布时间-TODO 将来基于延迟队列实现延迟发布
                wmNews.setStatus(MediaEnum.NEWS_SUBMIT_STATUS_PASS.getVal());
                newsService.updateById(wmNews);

                //TODO 延迟任务消息生产者
                //计算延迟消息时间
                long time = wmNews.getPublishTime().getTime();
                long delayTime = time - System.currentTimeMillis();
                RBlockingDeque<Object> blockingDeque = redissonClient.getBlockingDeque("article-publish");
                RDelayedQueue<Object> delayedQueue = redissonClient.getDelayedQueue(blockingDeque);
                delayedQueue.offer(wmNews.getId().toString(), delayTime, TimeUnit.MILLISECONDS);
            }
        }
    }


    /**
     * 基于自媒体文章对象构建文章对象，远程调用文章微服务保存文章（得到文章ID）；更新自媒体文章wm_news表中逻辑外键：article_id 文章ID
     *
     * @param wmNews 自媒体文章对象
     */
    private void remoteSaveArticle(WmNews wmNews) {
        //1.构建发布文章实体类对象ArticleDto
        ArticleDto articleDto = new ArticleDto();
        //1.1 封装基本信息 从自媒体文章拷贝到发布文章对象中 封面，标签 直接复制即可
        BeanUtils.copyProperties(wmNews, articleDto);

        //1.2 设置文章作者信息
        //1.2.1 根据自媒体文章中自媒体用户ID得到自媒体用户
        WmUser wmUser = wmUserService.getById(wmNews.getUserId());
        //1.2.2 通过自媒体用户得到作者ID
        Integer apAuthorId = wmUser.getApAuthorId();
        articleDto.setAuthorId(apAuthorId);
        articleDto.setAuthorName(wmUser.getName());

        //1.3 设置发布文章频道信息 根据自媒体对象中频道ID查询
        WmChannel wmChannel = channelService.getById(wmNews.getChannelId());
        if (wmChannel != null) {
            articleDto.setChannelId(wmChannel.getId());
            articleDto.setChannelName(wmChannel.getName());
        }
        //1.4 设置文章布局（自媒体中封面类型）
        articleDto.setLayout(wmNews.getType());

        //1.5 设置文章标记 刚发布为普通文章
        articleDto.setFlag(MediaEnum.ARTICLE_FLAG_NORMAL.getVal());

        //2. 远程调用文章微服务发布文章（保存或更新）
        ResponseResult<Long> responseResult = articleFeign.saveArticle(articleDto);
        if (responseResult.getCode().equals(AppHttpCodeEnum.SUCCESS.getCode())) {
            //3. 更新自媒体文章表中逻辑外键 发布文章ID
            Long articleId = responseResult.getData();
            wmNews.setArticleId(articleId);
            wmNews.setStatus(MediaEnum.NEWS_SUBMIT_STATUS_PUBLISH.getVal());
            newsService.updateById(wmNews);
        }
    }


    /**
     * 阿里云图片审核:审核自媒体文章对象中：封面，正文中图片
     *
     * @param images 封面+正文中图片
     * @param wmNews 自媒体文章对象
     * @return
     */
    private boolean checkImage(List<String> images, WmNews wmNews) {
        boolean result = false;
        // 如果没有图片,不需要进行图片审核,直接返回true
        if (images.size() <= 0) {
            return true;
        }
        try {
            // 调用阿里云图片审核
            // Map map = imageScan.checkUrl(images);    // 如果minio部署在外网服务器,可以直接通过URL地址来审核
            // 先将图片从minio服务器下载下来,然后调用阿里云进行审核
            List<byte[]> imageList = new ArrayList<>();
            for (String url : images) {
                // 下载图片
                InputStream inputStream = minIOService.download(url);
                byte[] bytes = IOUtils.toByteArray(inputStream);
                imageList.add(bytes);
            }
            Map map = aliyunGreenService.imageScan(imageList);
            String suggestion = (String) map.get("suggestion");
            // 判断结果
            switch (suggestion) {
                case "pass":
                    // 审核通过
                    result = true;
                    break;
                case "block":
                    // 审核不通过
                    wmNews.setStatus(NEWS_SUBMIT_STATUS_FAIL.getVal());
                    // 获取不通过的原因
                    String label = (String) map.get("label");
                    wmNews.setReason("阿里云图片审核不通过: " + label);
                    newsService.updateById(wmNews);
                    break;
                case "review":
                    // 人工审核
                    wmNews.setStatus(NEWS_SUBMIT_STATUS_ARTIFICIAL.getVal());
                    newsService.updateById(wmNews);
                    break;
            }
        } catch (Exception e) {
            e.printStackTrace();
            throw new RuntimeException("审核图片失败");
        }
        return result;
    }

    /**
     * 阿里云文本审核
     *
     * @param text   文章文本
     * @param wmNews 自媒体文章对象
     * @return
     */
    private boolean checkText(String text, WmNews wmNews) {
        boolean result = false;
        try {
            Map map = aliyunGreenService.greenTextScan(text);
            String suggestion = (String) map.get("suggestion");
            // 判断结果
            switch (suggestion) {
                case "pass":
                    // 审核通过
                    result = true;
                    break;
                case "block":
                    // 审核不通过
                    wmNews.setStatus(NEWS_SUBMIT_STATUS_FAIL.getVal());
                    // 获取不通过的原因
                    String label = (String) map.get("label");
                    wmNews.setReason("阿里云文本审核不通过: " + label);
                    newsService.updateById(wmNews);
                    break;
                case "review":
                    // 人工审核
                    wmNews.setStatus(NEWS_SUBMIT_STATUS_ARTIFICIAL.getVal());
                    newsService.updateById(wmNews);
                    break;
            }
        } catch (Exception e) {
            e.printStackTrace();
            throw new RuntimeException("审核内容失败");
        }
        return result;
    }

    /**
     * 从文章内容中提取文本和图片
     *
     * @param content
     * @return
     */
    private Map<String, Object> getTextAndImageFromContent(String content) {
        Map<String, Object> mapResult = new HashMap<>();
        //1.定义文本返回结果
        StringBuilder sb = new StringBuilder();
        //2.定义图片集合
        List<String> images = new ArrayList<>();

        //3.将文章内容转为集合对象
        List<ContentDto> contentDtos = JSON.parseArray(content, ContentDto.class);

        //4.遍历集合获取文本以及图片
        for (ContentDto dto : contentDtos) {
            if ("text".equals(dto.getType())) {
                //拼接文本字符串
                sb.append(dto.getValue());
            } else {
                //图片集合
                images.add(dto.getValue());
            }
        }
        mapResult.put("text", sb.toString());
        mapResult.put("images", images);
        return mapResult;
    }


    @Autowired
    private IWmSensitiveService sensitiveService;


    /**
     * 检查传入的文本是否包含敏感词
     *
     * @param text   文本
     * @param wmNews 自媒体用户发布的文章
     * @return
     */
    private boolean checkSensitive(String text, WmNews wmNews) {
        boolean result = false;
        //1.判断敏感词工具中的map是否为空，如果为空，从数据库中加载敏感词到Map中
        if (CollectionUtils.isEmpty(SensitiveWordUtil.dictionaryMap)) {
            LambdaQueryWrapper<WmSensitive> queryWrapper = new LambdaQueryWrapper<>();
            queryWrapper.select(WmSensitive::getSensitives);
            //1.1 查询敏感词表中名称列
            List<WmSensitive> sensitiveList = sensitiveService.list(queryWrapper);
            if (!CollectionUtils.isEmpty(sensitiveList)) {
                List<String> collect = sensitiveList.stream().map(WmSensitive::getSensitives).collect(Collectors.toList());
                //1.2 初始化敏感词map
                SensitiveWordUtil.initMap(collect);
            }
        }
        //2. 审核标题
        Map<String, Integer> mapTitle = SensitiveWordUtil.matchWords(wmNews.getTitle());
        if (mapTitle.size() <= 0) {
            //2.1 文章标题审核通过
            //2.1.1 继续审核文章内容是否包含敏感词
            Map<String, Integer> mapContent = SensitiveWordUtil.matchWords(text);
            if (CollectionUtils.isEmpty(mapContent)) {
                //内容审核不包含敏感词
                result = true;
            } else {
                //2.1.3 内容审核失败
                wmNews.setStatus(NEWS_SUBMIT_STATUS_FAIL.getVal());
                Set<String> keySet = mapContent.keySet();
                String join = String.join(",", keySet);
                wmNews.setReason("内容中包含敏感词：" + join);
                newsService.updateById(wmNews);
            }
        } else {
            //2.2 文章标题审核失败
            wmNews.setStatus(NEWS_SUBMIT_STATUS_FAIL.getVal());
            Set<String> keySet = mapTitle.keySet();
            String join = String.join(",", keySet);
            wmNews.setReason("标题中包含敏感词：" + join);
            newsService.updateById(wmNews);
            return false;
        }
        return result;
    }
}
